[Bugfix] Fix incorrect updates to num_computed_tokens in multi-step scheduling #9038
+179
−110
We went looking everywhere, but couldn’t find those commits.
Sometimes commits can disappear after a force-push. Head back to the latest changes here.